package com.zxlfysj.tree;

/**
 * 以数组顺序存储二叉树
 *
 * @author yangshujing
 * @create 2020-08-19 20:29
 */
public class ArrayBinaryTreeDemo {
    public static void main(String[] args) {
        int[] arr = {1, 2, 3, 4, 5, 6, 7};
        ArrayBinaryTree arrayBinaryTree = new ArrayBinaryTree(arr);
        //测试前序遍历
        arrayBinaryTree.preOrder(); //1 2 4 5 3 6 7
        System.out.println();
        //测试中序遍历
        arrayBinaryTree.infixOrder(); //4 2 5 1 6 3 7
        System.out.println();
        //测试后序遍历
        arrayBinaryTree.postOrder(); //4 5 2 6 7 3 1
    }
}

class ArrayBinaryTree {
    private int[] arr;

    public ArrayBinaryTree(int[] arr) {
        this.arr = arr;
    }

    //重载前序遍历
    public void preOrder() {
        this.preOrder(0);
    }

    //重载中序遍历
    public void infixOrder() {
        this.infixOrder(0);
    }

    //重载后序遍历
    public void postOrder() {
        this.postOrder(0);
    }

    /**
     * 前序遍历
     *
     * @param index 数组的下标
     */
    public void preOrder(int index) {
        if (arr == null || arr.length == 0) {
            System.out.println("数组为空，不能遍历");
            return;
        }
        //输出当前这个元素
        System.out.print(arr[index] + " ");
        //向左递归遍历
        if ((index * 2 + 1) < arr.length) {
            preOrder(2 * index + 1);
        }
        //向右递归遍历
        if ((index * 2 + 2) < arr.length) {
            preOrder(2 * index + 2);
        }
    }

    /**
     * 中序遍历
     *
     * @param index 数组的下标
     */
    public void infixOrder(int index) {
        if (arr == null || arr.length == 0) {
            System.out.println("数组为空，不能遍历");
            return;
        }
        //向左递归遍历
        if ((index * 2 + 1) < arr.length) {
            infixOrder(2 * index + 1);
        }
        //输出当前这个元素
        System.out.print(arr[index] + " ");
        //向右递归遍历
        if ((index * 2 + 2) < arr.length) {
            infixOrder(2 * index + 2);
        }
    }

    /**
     * 后序遍历
     *
     * @param index 数组的下标
     */
    public void postOrder(int index) {
        if (arr == null || arr.length == 0) {
            System.out.println("数组为空，不能遍历");
            return;
        }
        //向左递归遍历
        if ((index * 2 + 1) < arr.length) {
            postOrder(2 * index + 1);
        }
        //向右递归遍历
        if ((index * 2 + 2) < arr.length) {
            postOrder(2 * index + 2);
        }
        //输出当前这个元素
        System.out.print(arr[index] + " ");
    }
}
